Fixed some testing stuff

Signed-off-by: hajee <hajee@moretIA.com>