Fixing testing error