Fixed the solution

Signed-off-by: Sovak <sovak007@gmail.com>