Wrapping callbacks with @app makes functions return NoneType for pytests

I am trying to write pyttests for my application. When I debug, I can see that the imported functions from my callbacks files are of type NoneType. If i wrap those with regulas @callback from import dash then it works, but my application does not. This is probably because I am use dash-extensions which needs me to wrap my callbacks with @app.callback since it needs to do magic with the transform features (@Emil)

Has anyone seen behavior like this and can suggest a solution?