Tests now somewhat work

Signed-off-by: Jeremy Ho <jujaga@gmail.com>