Fixes testing error