Fixes test error

Fixes a test error

Author: Shockolate