fixing testing bug