Simplified code

Signed-off-by: K-ballo <k@fusionfenix.com>