Fix tests

Signed-off-by: Ian Macalinao <me@ian.pw>