Fixed the running test