Fixade kod för testfall

Signed-off-by: Anton <anton_peace@hotmail.com>