Fix last commit (forgot to rebase)

Signed-off-by: Aleksi Sapon <QcTechs@gmail.com>