Fixed test related to code.

Signed off by: Daniel Alves <danpaulalves@gmail.com>