Add test command back in