fix(tests): Fixes tests at the end.