Simplified return expression

Signed-off-by: yukipuki <Dawid@Geek>