Fixed a type on test