Reverted the tests to working ones

Signed-off-by: Daniel Koo <danieljamin.koo@gmail.com>