Fix with last commit

Signed-off-by: Javelin <javelinanddart@gmail.com>