Fixed testing bug