Fix for failing test

Fixes the test with correct methods.

Signed-off-by: Gorkem Ercan <gorkem.ercan@gmail.com>