Fixes

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