Fix last commit, now the code can compile

Signed-off-by: eldr4d <nikofinas@gmail.com>