Refactored tests to specify the running migrations through a decorator