Small fixes

Signed-off-by: Bartosz Domin <bdomin93@gmail.com>